package soft863.com.demo1007;

import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Partitioner;

public class FlowPartition extends Partitioner<Text, Flow> {
    @Override
    public int getPartition(Text text, Flow flow, int i) {
        String numb = text.toString().substring(0, 3);

        if ("134".equals(numb)) {
            return 0;
        } else if ("135".equals(numb)) {
            return 1;
        } else if ("136".equals(numb)) {
            return 2;
        } else if ("137".equals(numb)) {
            return 3;
        } else if ("138".equals(numb)) {
            return 4;
        } else if ("139".equals(numb)) {
            return 5;
        } else if ("150".equals(numb)) {
            return 6;
        }
        return 7;

    }
}
